python for data science week 3 solutions